# 作者：位微
# 你不是在写代码而是在解决问题
# 2025年09月17日17时42分34秒
# 2062873410@qq.com
import torch

a=torch.arange(0,12).view(3,4)
print('a:',a)
out=torch.chunk(a,2,dim=1)
print(out[0],out[0].shape)# 依然是动哪个轴哪个轴发生改变
print(out[1],out[1].shape)

a=torch.arange(0,12).view(3,4)
out=torch.split(a,[1,1,2],dim=1)
for t in out:
    print(t,t.shape)